Python Unit Tests using PyTest